matplotlib == 3.4.3
numpy == 1.21.3
scipy == 1.7.1
torch == 1.10.0+cu113
